'''
for test
'''

from airflow.models import DagPickle
from airflow.models.idata_task import Task
from airflow.utils.session import create_session

task = Task.get_task(task_id=1)
operator = task.create_operator()

with create_session() as session:
    pickle = DagPickle(operator.dag)
    pickle.save()
    print(pickle.id)
